fairnesscoop / permacoop

import { Customer } from '../Customer/Customer.entity';

export enum InvoiceUnits {

  DAY = 'day',

  HOUR = 'hour'

}

@Entity()

export class Project {

  @PrimaryGeneratedColumn('uuid')

  private id: string;

  @Column({ type: 'varchar', nullable: false })

  private name: string;

  @Column('enum', { enum: InvoiceUnits, nullable: false })

  private invoiceUnit: InvoiceUnits;

  @Column({ type: 'timestamp', default: () => 'CURRENT_TIMESTAMP' })

  private createdAt: Date;

  @ManyToOne(type => Customer, { nullable: false, onDelete: 'CASCADE' })

  private customer: Customer;

  constructor(

    name: string,

    invoiceUnit: InvoiceUnits,

    customer: Customer

  ) {

    this.name = name;

    this.invoiceUnit = invoiceUnit;

    this.customer = customer;

  }

  public getId(): string {

    return this.id;

  }

  public getName(): string {

    return this.name;

  }

  public getInvoiceUnit(): InvoiceUnits {

    return this.invoiceUnit;

  }

  public getCreatedAt(): Date {

    return this.createdAt;

  }

  public getCustomer(): Customer {

    return `[${this.customer.getName()}] ${this.getName()}`;

  }

  public update(

    customer: Customer,

    invoiceUnit: InvoiceUnits,

    name: string

  ): void {

    this.customer = customer;

    this.invoiceUnit = invoiceUnit;

    this.name = name;

  }

}